from wordcloud import WordCloud
import matplotlib.pyplot as plt

# 初始化一个空列表存储文本数据
data = []

# 逐行读取 JSON 文件的文本数据
with open('data.json', 'r', encoding='utf-8') as file:
    for line in file:
        data.append(line.strip())  # 去除每行文本开头和结尾的空白字符

# 将文本数据合并成一个长字符串
text_data = ' '.join(data)

# 指定中文字体文件路径
font_path = 'msyh.ttc'  # 替换为您下载的中文字体文件路径

# 创建词云对象并生成词云图
wordcloud = WordCloud(font_path=font_path, width=800, height=400, background_color='white').generate(text_data)

# 显示词云图
plt.figure(figsize=(10, 5))
plt.imshow(wordcloud, interpolation='bilinear')
plt.axis('off')

# 保存词云图为 word-cloud.png 文件
plt.savefig('word-cloud.png')

# 显示词云图
plt.show()
